### 2. 学生信息管理系统  id 、name、age、address  （id不能重复）
"""
1、添加学生信息
2、修改学生信息
3、删除学生信息
4、查看当前所有学生信息
5、退出
"""
# 学生信息管理系统
student_data = []

def add_student(id, name, age, address):
    for student in student_data:
        if student['id'] == id:
            print("学生ID已存在，请重新输入。")
            return
    student_data.append({'id': id, 'name': name, 'age': age, 'address': address})
    print("学生信息添加成功。")


def update_student(id, name, age, address):
    if not student_data:
        print("无学生信息，无法修改信息。")
    for student in student_data:
        if student['id'] == id:
            student['name'] = name
            student['age'] = age
            student['address'] = address
            print("学生信息更新成功。")
            return
    print("找不到学生ID，无法更新信息。")


def delete_student(id):
    if not student_data:
        print("无学生信息，无法删除。")
    for student in student_data:
        if student['id'] == id:
            student_data.remove(student)
            print("学生信息删除成功。")
            return
    print("找不到学生ID，无法删除信息。")


def view_students():
    if not student_data:
        print("当前没有学生信息。")
    else:
        print("当前所有学生信息：")
        for student in student_data:
            print(f"ID: {student['id']}, 姓名: {student['name']}, 年龄: {student['age']}, 地址: {student['address']}")

def inputinfo(flag):
    if flag == "1":
        id = input("请输入学生ID：")
        name = input("请输入学生姓名：")
        age = input("请输入学生年龄：")
        address = input("请输入学生地址：")
        return id, name, age, address
    elif flag == "2":
        id = input("请输入要修改的学生ID：")
        name = input("请输入新的学生姓名：")
        age = input("请输入新的学生年龄：")
        address = input("请输入新的学生地址：")
        return id, name, age, address


def main():
    while True:
        print("学生信息管理系统")
        print("1、添加学生信息\n2、修改学生信息\n3、删除学生信息\n4、查看当前所有学生信息\n5、退出")
        choice = input("请选择操作（1/2/3/4/5）：")
        if choice == '1':
            # 输入基本信息
            id,name,age,address = inputinfo(choice)
            add_student(id, name, age, address)
        elif choice == '2':
            # 输入基本信息
            id,name,age,address = inputinfo(choice)
            update_student(id, name, age, address)
        elif choice == '3':
            id = input("请输入要删除的学生ID：")
            delete_student(id)
        elif choice == '4':
            # 查看所有学生信息
            view_students()
        elif choice == '5':
            print("退出学生信息管理系统。")
            break
        else:
            print("无效的选择，请重新输入。")

if __name__ == "__main__":
    main()